// ============================================================================
//
// Copyright (C) 2006-2007 Dengues
//
// Google Group: http://groups.google.com/group/dengues
// QQ Group: 24885404
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
//
// ============================================================================
package org.dengues.designer.ui.process.policies;

import org.dengues.designer.ui.process.commands.CompConnectionCreateCommand;
import org.dengues.designer.ui.process.commands.CompConnectionReconnectionCommand;
import org.dengues.designer.ui.process.models.CompConnection;
import org.dengues.designer.ui.process.models.CompNode;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.editpolicies.GraphicalNodeEditPolicy;
import org.eclipse.gef.requests.CreateConnectionRequest;
import org.eclipse.gef.requests.ReconnectRequest;

/**
 * Qiang.Zhang.Adolf@gmail.com class global comment. Detailled comment <br/>
 * 
 * @since 2007-10-9 下午11:14:54
 */
public class CompNodeGraphicalEditPolicy extends GraphicalNodeEditPolicy {

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.editpolicies.GraphicalNodeEditPolicy#getConnectionCompleteCommand(org.eclipse.gef.requests.CreateConnectionRequest)
     */
    @Override
    protected Command getConnectionCompleteCommand(CreateConnectionRequest request) {
        CompConnectionCreateCommand cmd = (CompConnectionCreateCommand) request.getStartCommand();
        cmd.setTarget((CompNode) getHost().getModel());
        return cmd;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.editpolicies.GraphicalNodeEditPolicy#getConnectionCreateCommand(org.eclipse.gef.requests.CreateConnectionRequest)
     */
    @Override
    protected Command getConnectionCreateCommand(CreateConnectionRequest request) {
        CompNode node = (CompNode) getHost().getModel();
        CompConnectionCreateCommand cmd = new CompConnectionCreateCommand(node);
        request.setStartCommand(cmd);
        return cmd;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.editpolicies.GraphicalNodeEditPolicy#getReconnectSourceCommand(org.eclipse.gef.requests.ReconnectRequest)
     */
    @Override
    protected Command getReconnectSourceCommand(ReconnectRequest request) {
        CompConnection conn = (CompConnection) request.getConnectionEditPart().getModel();
        CompNode newSource = (CompNode) getHost().getModel();
        CompConnectionReconnectionCommand command = new CompConnectionReconnectionCommand(conn);
        command.setNewTarget(newSource);
        return command;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.editpolicies.GraphicalNodeEditPolicy#getReconnectTargetCommand(org.eclipse.gef.requests.ReconnectRequest)
     */
    @Override
    protected Command getReconnectTargetCommand(ReconnectRequest request) {
        CompConnection conn = (CompConnection) request.getConnectionEditPart().getModel();
        CompNode newSource = (CompNode) getHost().getModel();
        CompConnectionReconnectionCommand command = new CompConnectionReconnectionCommand(conn);
        command.setNewSource(newSource);
        return command;
    }

}
